package me.helllp.book.netty.step09;

import io.netty.bootstrap.ServerBootstrap;
import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioServerSocketChannel;
import io.netty.handler.codec.http.HttpObjectAggregator;
import io.netty.handler.codec.http.HttpRequestDecoder;
import io.netty.handler.codec.http.HttpResponseEncoder;
import io.netty.handler.stream.ChunkedWriteHandler;

public class HttpFileServer {
	private static final String DEFAULT_URL = "/src/main/java";
	
	private void run(final int port, final String url) throws Exception {
		EventLoopGroup bossGroup = new NioEventLoopGroup();
		EventLoopGroup workerGroup = new NioEventLoopGroup();
		
		try {
			ServerBootstrap b = new ServerBootstrap();
			b.group(bossGroup, workerGroup)
				.channel(NioServerSocketChannel.class)
				.childHandler(new ChannelInitializer<SocketChannel>(){

					@Override
					protected void initChannel(SocketChannel ch) throws Exception {
						ch.pipeline()
							.addLast("http-decoder", new HttpRequestDecoder())
							.addLast("http-aggregator", new HttpObjectAggregator(65536))
							.addLast("http-encoder", new HttpResponseEncoder())
							.addLast("http-chunked", new ChunkedWriteHandler())
							.addLast("fileServerHandler", new HttpFileServerHandler(url));
					}
					
				});
			ChannelFuture future = b.bind("localhost", port).sync();
			System.out.println("HTTP 文件目录服务器启动！端口为" + port);
			future.channel().closeFuture().sync();
		} finally {
			bossGroup.shutdownGracefully();
			workerGroup.shutdownGracefully();
		} 
	}
	
	public static void main(String[] args) throws Exception {
		int port = 8080;
		
		new HttpFileServer().run(port, DEFAULT_URL);
	}
}
